package com.study.sbsummary.utils;

import javax.crypto.Mac;
import javax.crypto.spec.SecretKeySpec;
import java.util.Base64;

public class HMACSHA1 {
    private static final String HMAC_SHA1 = "HmacSHA1";
    private static String JINSHAN_AK = "/BuBx5t7pKxexhua31BW";  //AccessKey
    private static String JINSHAN_SK = "aEMoLuatE1hUaXVQThry1/NP84iJX1gkUToadqHT";

    /**
     * 使用 HMAC-SHA1 签名算法对data进行签名
     *
     * @param data 被签名的字符串
     * @param key  密钥
     * @return 加密后的字符串
     */
    public static String getSignature(String data, String key) throws Exception {
        //根据给定的字节数组构造一个密钥,第二参数指定一个密钥算法的名称
        SecretKeySpec signinKey = new SecretKeySpec(key.getBytes(), HMAC_SHA1);
        //生成一个指定 Mac 算法 的 Mac 对象
        Mac mac = Mac.getInstance(HMAC_SHA1);
        //用给定密钥初始化 Mac 对象
        mac.init(signinKey);
        //完成 Mac 操作
        byte[] rawHmac = mac.doFinal(data.getBytes());
//        return byte2hex(rawHmac);
        return encodeWithBase64(rawHmac);
    }

    private static String byte2hex(final byte[] b) {
        String hs = "";
        String stmp = "";
        for (int n = 0; n < b.length; n++) {
            // 以十六进制（基数 16）无符号整数形式返回一个整数参数的字符串表示形式。
            stmp = (Integer.toHexString(b[n] & 0xFF));
            if (stmp.length() == 1) {
                hs = hs + "0" + stmp;
            } else {
                hs = hs + stmp;
            }
        }
        return hs;
    }

    public static String encodeWithBase64(byte[] hmacSHA1) throws Exception {
        return Base64.getEncoder().encodeToString(hmacSHA1);
    }
}
